package com.xiaoxiaoniu.airoleplay.repository;

import com.xiaoxiaoniu.airoleplay.model.Conversation;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Optional;

/**
 * 会话数据访问接口
 * 提供对会话实体的数据库操作方法
 */
@Repository
public interface ConversationRepository extends JpaRepository<Conversation, Long> {

    /**
     * 根据会话ID查找会话
     * 
     * @param conversationId 会话ID
     * @return 包含会话的Optional对象
     */
    Optional<Conversation> findByConversationId(String conversationId);

    /**
     * 根据角色名称查找会话，并按更新时间降序排列
     * 
     * @param roleName 角色名称
     * @return 会话列表
     */
    List<Conversation> findByRoleNameOrderByUpdatedAtDesc(String roleName);

    /**
     * 查找所有会话并按更新时间降序排列
     * 
     * @return 会话列表
     */
    @Query("SELECT c FROM Conversation c ORDER BY c.updatedAt DESC")
    List<Conversation> findAllOrderByUpdatedAtDesc();

    /**
     * 统计指定角色的会话数量
     * 
     * @param roleName 角色名称
     * @return 会话数量
     */
    @Query("SELECT COUNT(c) FROM Conversation c WHERE c.roleName = :roleName")
    long countByRoleName(@Param("roleName") String roleName);
    
    /**
     * 查找超过指定时间没有更新的会话
     * 
     * @param thresholdTime 时间阈值
     * @return 过期会话列表
     */
    @Query("SELECT c FROM Conversation c WHERE c.updatedAt < :thresholdTime")
    List<Conversation> findExpiredConversations(@Param("thresholdTime") LocalDateTime thresholdTime);
    
    /**
     * 获取所有会话ID
     * 
     * @return 会话ID列表
     */
    @Query("SELECT c.conversationId FROM Conversation c")
    List<String> findAllConversationIds();
}